import torch


# 随机生成数据
data = torch.randint(0, 10, [4, 5])
print(data)
'''
结果为：
tensor([[1, 9, 1, 2, 1],
        [7, 3, 5, 6, 5],
        [2, 7, 5, 6, 9],
        [1, 2, 0, 3, 5]])

'''

# todo 简单行， 列索引
print(data[0])
# 第一行数据： [1, 9, 1, 2, 1]
print(data[:, 0])
# 第一列数据： [1, 7, 2, 1]
# todo 列表索引
print(data[[0, 1], [1, 2]])
# 结果为： tensor([9, 5])
print(data[[[0], [1]], [1, 2]])
'''
输出结果：
tensor([[9, 1], 
[3, 5]])
'''

# todo 范围索引
print(data[:3, :2])    # 前三行的前两列数据

print(data[2:, :2])  # 第二行到最后的前两列数据


